from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
ax = fig.gca(projection='3d')
X,Y = np.meshgrid(np.arange(10),np.arange(5)) 

Z = np.zeros((5,10))  
Z[0][0]=-0.5
Z[0][1]=-0.5
Z[1][0]=-0.5
Z[-1][-1]=0.5
Z[2][2]=0.8
Z[2][3]=0.8


for i in X, Y, Z:
    print(i.shape)

#X, Y, Z = axes3d.get_test_data(0.05)
#ax.plot_surface(X, Y, Z, rstride=8, cstride=10, alpha=0.3)
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, alpha=0.3)
#cset = ax.contour(X, Y, Z, zdir='z', offset=-1)
#cset = ax.contour(X, Y, Z, zdir='x', offset=-4)
#cset = ax.contour(X, Y, Z, zdir='y', offset=-1)

ax.set_xlabel('X')
#ax.set_xlim(-40, 40)
ax.set_ylabel('Y')
#ax.set_ylim(-40, 40)
ax.set_zlabel('Z')
ax.set_zlim(-1, 1)

plt.show()

